how to turn 2d array into 1d array python